package Leetcode.图;

import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @ClassName 钥匙和房间
 * @since: 2023/8/21 14:06
 * @auth: kirito
 * @description:
 * 有 n 个房间，房间按从 0 到 n - 1 编号。
 * 最初，除 0 号房间外的其余所有房间都被锁住。
 * 你的目标是进入所有的房间。然而，你不能在没有获得钥匙的时候进入锁住的房间。
 *
 * 当你进入一个房间，你可能会在里面找到一套不同的钥匙，
 * 每把钥匙上都有对应的房间号，即表示钥匙可以打开的房间。
 * 你可以拿上所有钥匙去解锁其他房间。
 *
 * 给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。
 * 如果能进入 所有 房间返回 true，否则返回 false。
 *
 * 示例 1：
 *
 * 输入：rooms = [[1],[2],[3],[]]
 * 输出：true
 * 解释：
 * 我们从 0 号房间开始，拿到钥匙 1。
 * 之后我们去 1 号房间，拿到钥匙 2。
 * 然后我们去 2 号房间，拿到钥匙 3。
 * 最后我们去了 3 号房间。
 * 由于我们能够进入每个房间，我们返回 true。
 * 示例 2：
 *
 * 输入：rooms = [[1,3],[3,0,1],[2],[0]]
 * 输出：false
 * 解释：我们不能进入 2 号房间。
 *
 **/
public class 钥匙和房间 {
    boolean[] vis; // 记录房间是否被访问过的数组
    int num; // 记录已经访问的房间数量
//    roomCount 应该作为引用传递给 dfs 方法，这样它才能正确地更新。
//    在 Java 中，基本类型（如 int）是按值传递的，这意味着在 dfs 方法中对 roomCount 的任何修改
//    都不会影响原始的 roomCount 变量。

    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        int n = rooms.size(); // 房间的总数
        num = 0;
        vis = new boolean[n]; // 初始化房间访问数组
        dfs(rooms, 0); // 从第一个房间开始进行深度优先搜索
        return num == n; // 判断是否所有房间都被访问过
    }

    private void dfs(List<List<Integer>> rooms, int i) {
        vis[i] = true; // 将当前房间标记为已访问
        num++; // 已访问的房间数量加1
        for (int it : rooms.get(i)) { // 遍历当前房间的钥匙列表
            if (!vis[it]) { // 如果钥匙所对应的房间未被访问过
                dfs(rooms, it); // 继续对该房间进行深度优先搜索
            }
        }
    }

    public boolean canVisitAllRooms2(List<List<Integer>> rooms) {
        int n = rooms.size(); // 房间的总数
        boolean[] visited = new boolean[n]; // 记录房间是否被访问过的数组
        visited[0] = true; // 将第一个房间标记为已访问

        Queue<Integer> queue = new LinkedList<>(); // 创建一个队列存储待访问的房间
        queue.offer(0); // 将第一个房间加入队列

        int num = 1; // 记录已经访问的房间数量，初始为1

        while (!queue.isEmpty()) {
            int currRoom = queue.poll(); // 取出队列中的当前房间

            for (int key : rooms.get(currRoom)) {
                if (!visited[key]) { // 如果钥匙所对应的房间未被访问过
                    visited[key] = true; // 将该房间标记为已访问
                    queue.offer(key); // 将该房间加入队列
                    num++; // 已访问的房间数量加1
                }
            }
        }

        return num == n; // 判断是否所有房间都被访问过
    }
}
